from .module import PREPROCESS
import cv2
import numpy as np


@PREPROCESS.register_module
class RandomFlip:

    def __init__(self):
        # self.is_flip = np.random.randint(2)
        self.is_flip = None

    def process(self, img, anno):
        anno = anno.copy()
        self.is_flip = np.random.randint(2)
        if self.is_flip > 0:
            img = cv2.flip(img, 1)
        h, w, c = img.shape
        anno[1] = w-anno[3]
        anno[2] = h-anno[4]
        anno[3] = w-anno[1]
        anno[4] = h-anno[2]
        return img, anno